from flask_security.forms import RegisterForm, Required, LoginForm
from wtforms import StringField, SubmitField


class ExtendedRegisterForm(RegisterForm):
    loginame = StringField('login_name', [Required()])
    name = StringField('name', [Required()])


class ExtendedLoginForm(LoginForm):
    email = StringField('Email or Login_name', validators=[Required()])
    submit = SubmitField('login')
